#!/usr/bin/env python
# -*- coding: utf-8 -*-
import redis

redis_db = redis.StrictRedis(host='localhost', port=6379, db=0, charset='utf8')


def set_phone_captcha(phone, captcha):
    redis_db.set(phone, captcha)
    redis_db.expire(phone, 600)  # 设置过期时间，10分钟


def verify_phone_captcha(phone, captcha):
    # 当拿不到 phone 这个key的时候
    if not redis_db.get(phone):
        return False
    value = redis_db.get(phone).decode('utf-8')
    if captcha == value:
        return True
    return False
